#pragma once

#include <iostream>
#include <string>

#include <jsoncpp/json/json.h>
#include <signal.h>

#include "./compiler.hpp"
#include "./runner.hpp"

namespace ns_compile_run
{
    class compile_run
    {
    public:
        static void remove_temp_file(const std::string file_name)
        {
            // 在用户代码编译运行的过程中,我们不知道究竟生成了多少个文件
            // 但是我们知道会生成哪些文件,所以我们把所有的文件都删除了就行了
            std::string src = ns_util::path_util::src(file_name);
            if (ns_util::file_util::if_file_exists(src))
            {
                // 这个函数Linux下是用来删除文件的
                unlink(src.c_str());
            }

            std::string compiler_error = ns_util::path_util::compiler_error(file_name);
            if (ns_util::file_util::if_file_exists(compiler_error))
            {
                unlink(compiler_error.c_str());
            }

            std::string execute = ns_util::path_util::exe(file_name);
            if (ns_util::file_util::if_file_exists(execute))
            {
                unlink(execute.c_str());
            }

            std::string stdin = ns_util::path_util::stdin(file_name);
            if (ns_util::file_util::if_file_exists(stdin))
            {
                unlink(stdin.c_str());
            }

            std::string stdout = ns_util::path_util::stdout(file_name);
            if (ns_util::file_util::if_file_exists(stdout))
            {
                unlink(stdout.c_str());
            }

            std::string stderr = ns_util::path_util::stderr(file_name);
            if (ns_util::file_util::if_file_exists(stderr))
            {
                unlink(stderr.c_str());
            }
        }

        /**************************************************************
         * code > 0:进程收到了信号导致异常崩溃
         * code < 0:执行代码的过程中报错(用户代码为空,编译报错,内部错误等)
         * code = 0:整个过程全部完成
         *可以根据信号,然后不断地增加不同的情况
         ***************************************************************/
        static std::string code_to_desc(int code, const std::string &file_name)
        {
            std::string desc;
            switch (code)
            {
            case 0:
                desc = "代码编译运行成功";
                break;
            case -1:
                desc = "用户提交代码为空";
                break;
            case -2:
                desc = "内部错误";
                break;
            case -3:
                // 用户代码编译的时候发生了错误
                ns_util::file_util::read_file(ns_util::path_util::compiler_error(file_name), &desc, true);
                break;
            case SIGABRT: // 6
                desc = "程序内存超过了限制";
                break;
            case SIGXCPU: // 24
                desc = "程序运行时间超时";
                break;
            case SIGFPE: // 8
                desc = "程序浮点数溢出";
                break;
            default:
                break;
            }
            return desc;
        }

        /*****************************************************
         * in_json:
         * code:用户提交的代码
         * input:用户给自己提交的代码对应的输入,不做处理
         * cpu_limit:时间要求
         * mem_limit:空间要求
         *
         * out_json:
         * 必填:
         * status_code:状态码
         * reason:本次请求结果
         *
         * 选填:
         * stdout:用户程序运行完的结果
         * stderr:程序运行完的错误结果
         *******************************************************/
        static void start(const std::string &in_json, std::string *out_json)
        {
            // 反序列化
            Json::Value in_value;
            Json::Reader reader;
            reader.parse(in_json, in_value);

            std::string code = in_value["code"].asString();
            std::string input = in_value["input"].asString();
            int cpu_limit = in_value["cpu_limit"].asInt();
            int mem_limit = in_value["mem_limit"].asInt();

            // 因为下面要使用到goto语句,在goto语句的区间,是不能有变量进行初始化的
            // 所以我们这里提前把需要的变量进行初始化
            int status_code = 0;   // 状态码
            std::string file_name; // 用户代码形成的文件名
            int run_result = 0;    // 用户代码形成的程序运行之后的结果

            if (code.size() == 0)
            {
                // 说明用户传递过来的代码为空
                status_code = -1;
                goto END;
            }

            // 代码不为空,就可以形成代码对应的cpp文件了
            // 形成的文件名,我们需要形成具有唯一性的文件名,这个文件名不带有目录和文件后缀
            file_name = ns_util::file_util::unique_file_name();

            // 形成临时src文件
            if (!ns_util::file_util::write_file(ns_util::path_util::src(file_name), code))
            {
                status_code = -2; // 我们知道这里是形成临时文件失败,但是我们不用暴露给用户,我们只需要说明是内部错误就行了
                goto END;
            }

            // 对临时文件进行编译
            if (!ns_compiler::compiler::compile(file_name))
            {
                // 这里说明用户代码编译失败了
                status_code = -3;
                goto END;
            }

            // 对编译后的程序,进行执行
            run_result = ns_runner::runner::run(file_name, cpu_limit, mem_limit);
            if (run_result < 0)
            {
                status_code = -2; // 说明执行run函数的时候出错了,也是我们的内部错误
            }
            else if (run_result > 0)
            {
                // 说明用户的程序收到了信号,终止了运行
                status_code = run_result;
            }
            else
            {
                // 运行成功了
                status_code = 0;
            }
        END:
            Json::Value out_value;
            out_value["status_code"] = status_code;
            /**/ out_value["reason"] = code_to_desc(status_code, file_name);

            // 只有代码编译成功且正确运行的代码,才可以添加下面的字段
            if (status_code == 0)
            {
                std::string stdout;
                ns_util::file_util::read_file(ns_util::path_util::stdout(file_name), &stdout, true);
                out_value["stdout"] = stdout;

                std::string stderr;
                ns_util::file_util::read_file(ns_util::path_util::stderr(file_name), &stderr, true);
                out_value["stderr"] = stderr;
            }

            Json::FastWriter writer;
            *out_json = writer.write(out_value);

            // 移除生成的所有临时文件
            remove_temp_file(file_name);
        }
    };
}
